package com.sali.回溯算法;

/**
 * 编写一个程序，通过填充空格来解决数独问题。
 * <p>
 * 数独的解法需 遵循如下规则：
 * <p>
 * 数字 1-9 在每一行只能出现一次。
 * 数字 1-9 在每一列只能出现一次。
 * 数字 1-9 在每一个以粗实线分隔的 3x3 宫内只能出现一次。（请参考示例图）
 * 数独部分空格内已填入了数字，空白格用 '.' 表示。
 */
public class LeetCode_37 {

    public static void main(String[] args) {
        char[][] board = {
                {'5', '3', '.', '.', '7', '.', '.', '.', '.'},
                {'6', '.', '.', '1', '9', '5', '.', '.', '.'},
                {'.', '9', '8', '.', '.', '.', '.', '6', '.'},
                {'8', '.', '.', '.', '6', '.', '.', '.', '3'},
                {'4', '.', '.', '8', '.', '3', '.', '.', '1'},
                {'7', '.', '.', '.', '2', '.', '.', '.', '6'},
                {'.', '6', '.', '.', '.', '.', '2', '8', '.'},
                {'.', '.', '.', '4', '1', '9', '.', '.', '5'},
                {'.', '.', '.', '.', '8', '.', '.', '7', '9'}
        };

        solveSudoku(board);
        for (int i = 0; i < board.length; i++) {
            for (int j = 0; j < board[0].length; j++) {
                System.out.print(board[i][j] + " ");
            }
            System.out.println();
        }
    }


    public static void solveSudoku(char[][] board) {
        // corner case
        if (board == null) {
            return;
        }

        getRes(board, 0, 0);
    }

    private static boolean getRes(char[][] board, int row, int col) {
        if (row >= board.length) {
            return true;
        }
        if (col >= board[0].length) {
            return getRes(board, row + 1, 0);
        }
        if (board[row][col] != '.') {
            return getRes(board, row, col + 1);
        }

        for (char ch = '1'; ch <= '9'; ch++) {
            if (!isValid(board, row, col, ch)) {
                continue;
            }
            board[row][col] = ch;
            if (getRes(board, row, col + 1)) {
                return true;
            }
            board[row][col] = '.';
        }

        return false;
    }

    private static boolean isValid(char[][] board, int row, int col, char ch) {
        for (int i = 0; i < 9; i++) {
            if (board[i][col] == ch || board[row][i] == ch) {
                return false;
            }
            if (board[row / 3 * 3 + i / 3][col / 3 * 3 + i % 3] == ch) {
                return false;
            }
        }
        return true;
    }

}
